#pragma once

#include "txr_algo_dlm_v5_global.h"
#include "../txr_algo_dlm/txr_algo_dlm_impl.h"


namespace yolo_v6
{
	class YoloDet2;
}//namespace yolo_v6

using namespace yolo_v6;
namespace txr_algo_dlm
{
	class TxrAlgoDlmV5 : public TxrAlgoDlmImpl
	{
	public:
		TxrAlgoDlmV5();
		~TxrAlgoDlmV5();

		tk_bool Load(st_dlm_cfg & cfg);
		void Release();
		tk_int32 BatchSize();
		void Detect(st_detect_unit * p_unit, tk_int32 size);
	private:
		YoloDet2 * m_p_detect;
	};
}//namespace txr_algo_dlm